package com.aiobc.common.core.utils;
import com.aiobc.common.core.enums.SexEnum;
import org.springframework.util.Assert;
import org.springframework.util.StringUtils;

import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.time.LocalDate;
import java.util.*;
import java.util.regex.Pattern;

/**
 * com.aiobc.common.core.utils
 * 身份证强校验类
 *
 * @Author: liqiang
 * @Date: 2021/10/19
 * @Time: 13:08
 */
public class IDNumberUtil {
    /**
     * 从身份证号取生日
     * @param idNumber
     * @return
     */
    public static LocalDate getBirthday(String idNumber) {
        Assert.isTrue(StringUtils.hasText(idNumber),"身份证号不能为空！");
        String temporaryString = idNumber.length() == 18 ? idNumber.substring(0, 17) : idNumber.substring(0, 6) + "19" + idNumber.substring(6, 15);
        Assert.isTrue(isNumeric(temporaryString), "身份证15位号码都应为数字 ; 18位号码除最后一位外，都应为数字。");
        String birthdayStr = temporaryString.substring(6, 14);
        return DateUtils.parse(birthdayStr, DateUtils.YYYYMMDD);
    }

    /**
     * 从身份证号取性别
     * @param idNumber
     * @return
     */
    public static SexEnum getSex(String idNumber) {
        Assert.isTrue(StringUtils.hasText(idNumber),"身份证号不能为空！");
        idNumber = idNumber.substring(idNumber.length() - 2, idNumber.length() - 1);
        Assert.isTrue(isNumeric(idNumber), "身份证15位号码都应为数字 ; 18位号码除最后一位外，都应为数字。");
        return Integer.valueOf(idNumber) % 2 == SexEnum.FEMALE.ordinal() ? SexEnum.FEMALE : SexEnum.MALE;
    }

    /**
     * @param idNumber
     * @return
     * @throws ParseException
     */
    public static boolean validate(String idNumber) {
        Assert.isTrue(StringUtils.hasText(idNumber),"身份证号不能为空！");
        String[] valCodeArr = {"1", "0", "x", "9", "8", "7", "6", "5", "4", "3", "2"};
        String[] weightingFactor = {"7", "9", "10", "5", "8", "4", "2", "1", "6", "3", "7", "9", "10", "5", "8", "4", "2"};
        String temporaryString = "";
        /*================ 号码的长度 15位或18位 ================*/
        Assert.isTrue(idNumber.length() == 15 || idNumber.length() == 18, "身份证号码长度应该为15位或18位。");
        idNumber = idNumber.toLowerCase(Locale.ROOT);
        /*================ 数字 除最后以为都为数字 ================*/
        temporaryString = idNumber.length() == 18 ? idNumber.substring(0, 17) : idNumber.substring(0, 6) + "19" + idNumber.substring(6, 15);
        Assert.isTrue(isNumeric(temporaryString), "身份证15位号码都应为数字 ; 18位号码除最后一位外，都应为数字。");

        /*================ 出生年月是否有效 ================*/
        String strYear = temporaryString.substring(6, 10);
        String strMonth = temporaryString.substring(10, 12);
        String strDay = temporaryString.substring(12, 14);
        String dateString = strYear + "-" + strMonth + "-" + strDay;
        Assert.isTrue(isDate(dateString), "身份证生日无效。");

        GregorianCalendar gregorianCalendar = new GregorianCalendar();
        SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd");
        Date parse;
        try {
            parse = simpleDateFormat.parse(dateString);
        } catch (Exception e) {
            throw new RuntimeException("时间格式错误！");
        }
        Assert.isTrue((gregorianCalendar.get(Calendar.YEAR) - Integer.parseInt(strYear)) < 150 || (gregorianCalendar.getTime().getTime() - parse.getTime()) > 0, "身份证生日不在有效范围。");

        Assert.isTrue(Integer.parseInt(strMonth) < 12 || Integer.parseInt(strMonth) > 0, "身份证月份无效");
        Assert.isTrue(Integer.parseInt(strDay) < 31 || Integer.parseInt(strDay) > 0, "身份证日期无效");

        /*================ 地区码时候有效 ================*/
        Assert.isTrue(GetAreaCode().get(temporaryString.substring(0, 2)) != null, "身份证地区编码错误。");

        /*================ 判断最后一位的值 ================*/
        /*Y = mod(S, 11)   S = Sum(weightingFactor * valCodeArr)*/
        int S = 0;
        for (int i = 0; i < 17; i++) {
            S += Integer.parseInt(String.valueOf(temporaryString.charAt(i))) * Integer.parseInt(weightingFactor[i]);
        }
        int Y = S % 11;
        String strVerifyCode = valCodeArr[Y];
        temporaryString = temporaryString + strVerifyCode;

        Assert.isTrue(idNumber.length() == 18 && temporaryString.equals(idNumber), "身份证无效，不是合法的身份证号码");
        return true;
    }

    /**
     * 功能：设置地区编码
     *
     * @return Hashtable 对象
     */
    private static Hashtable<String, String> GetAreaCode() {
        Hashtable<String, String> hashtable = new Hashtable<>();
        hashtable.put("11", "北京");
        hashtable.put("12", "天津");
        hashtable.put("13", "河北");
        hashtable.put("14", "山西");
        hashtable.put("15", "内蒙古");
        hashtable.put("21", "辽宁");
        hashtable.put("22", "吉林");
        hashtable.put("23", "黑龙江");
        hashtable.put("31", "上海");
        hashtable.put("32", "江苏");
        hashtable.put("33", "浙江");
        hashtable.put("34", "安徽");
        hashtable.put("35", "福建");
        hashtable.put("36", "江西");
        hashtable.put("37", "山东");
        hashtable.put("41", "河南");
        hashtable.put("42", "湖北");
        hashtable.put("43", "湖南");
        hashtable.put("44", "广东");
        hashtable.put("45", "广西");
        hashtable.put("46", "海南");
        hashtable.put("50", "重庆");
        hashtable.put("51", "四川");
        hashtable.put("52", "贵州");
        hashtable.put("53", "云南");
        hashtable.put("54", "西藏");
        hashtable.put("61", "陕西");
        hashtable.put("62", "甘肃");
        hashtable.put("63", "青海");
        hashtable.put("64", "宁夏");
        hashtable.put("65", "新疆");
        hashtable.put("71", "台湾");
        hashtable.put("81", "香港");
        hashtable.put("82", "澳门");
        hashtable.put("91", "国外");
        return hashtable;
    }

    /**
     * 功能：判断字符串是否为数字
     *
     * @param str
     * @return
     */
    private static Pattern patternNumber = Pattern.compile("[0-9]*");

    private static boolean isNumeric(String str) {
        return patternNumber.matcher(str).matches();
    }

    /**
     * 功能：判断字符串是否为日期格式
     *
     * @param strDate
     * @return
     */
    private static Pattern patternDate = Pattern.compile("^((\\d{2}(([02468][048])|([13579][26]))[\\-\\/\\s]?((((0?[13578])|(1[02]))[\\-\\/\\s]?((0?[1-9])|([1-2][0-9])|(3[01])))|(((0?[469])|(11))[\\-\\/\\s]?((0?[1-9])|([1-2][0-9])|(30)))|(0?2[\\-\\/\\s]?((0?[1-9])|([1-2][0-9])))))|(\\d{2}(([02468][1235679])|([13579][01345789]))[\\-\\/\\s]?((((0?[13578])|(1[02]))[\\-\\/\\s]?((0?[1-9])|([1-2][0-9])|(3[01])))|(((0?[469])|(11))[\\-\\/\\s]?((0?[1-9])|([1-2][0-9])|(30)))|(0?2[\\-\\/\\s]?((0?[1-9])|(1[0-9])|(2[0-8]))))))");

    public static boolean isDate(String strDate) {
        return patternDate.matcher(strDate).matches();
    }
}
